package cn.itcast.web.servlet;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.Random;

/*
* 验证码:自己做一个验证码，然后通过获取图片的字节码数据 (对应register.html)
* 验证码逻辑:
* 1.在内存创建一个验证码图片对象代表图片
* BufferedImage image=new BufferedImage(宽,高,图片类型);
*
* 2.美化图片。(在画布中画图片的每一步都可以转化成代码，有对应的对象及方法)。
* //2.1.填充背景色
* //2.2.给图片画边框
* //2.3.在图片上写验证码
* //2.4.画干扰线
*
* 3.将内存中准备好的验证码图片输出到页面展示，(需要使用一些对象)
* ImageIO.write(图片对象，图片后缀，输出流对象)
* */
@WebServlet("/checkCodeServlet")
public class CheckCodeServlet extends HttpServlet {
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException,IOException{
        //1.创建一对象，在内存中代表图片(验证码图片对象),显示为100*50底色为黑色长方形
        int width=100;
        int height=50;
        BufferedImage image=new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);
        //2.美化图片(将默认的图片进行美化)
        //2.1.填充背景色
        Graphics g=image.getGraphics();//获取画笔对象
        g.setColor(Color.PINK);//设置画笔颜色
        g.fillRect(0,0,width,height);//填充一个矩形,从(0,0)起点的矩形
        //2.2.给图片画边框
        g.setColor(Color.BLUE);//设置画笔颜色为蓝色
        g.drawRect(0,0,width-1,height-1);//画一个以(0,0)起点，宽高的长方形
        //即在原有的长方形上画蓝色边框。(0,0)点指长方形的左上角。若宽高不减1则长方形的
        //边框只会出现一半，因为边框自身是有1px的宽度的，导致了另一半的边框被顶出去了，
        //所以需要宽高各让出1px给边框。
        String str="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
        //生成随机角标，(即随机获取一个不大于str.length()大小的整型)
        Random ran=new Random();
        //创建StringBuilder对象存放验证码上的字符，将来存放到session中实现数据共享
        //即可以通过session拿到验证码上的字符与用户输入的验证码字符进行比较。
        StringBuilder sb=new StringBuilder();
        for (int i=1;i<=4;i++){//在图片上动态生成4个验证码元素。
            int index=ran.nextInt(str.length());
            //获取字符
            char ch=str.charAt(index);//随机字符(获取str字符串中索引为index的字符)
            //将字符拼接到StringBuilder字符串对象中
            sb.append(ch);
            //2.3.在图片上写验证码
            g.drawString(ch+"",width/5*i,height/2);
        }
        //获取随机生成的验证码字符串
        String checkCode_session=sb.toString();
        //将验证码存入session中实现数据共享
        request.getSession().setAttribute("checkCode_session",checkCode_session);
        //2.4.画干扰线
        g.setColor(Color.GREEN);//设置画笔颜色
        //随机生成坐标点(x1,x2)和(y1,y2)两点
        for (int i=1;i<=10;i++){  //生成10条干扰线
            int x1=ran.nextInt(width);
            int x2=ran.nextInt(width);
            int y1=ran.nextInt(height);
            int y2=ran.nextInt(height);
            g.drawLine(x1,x2,y1,y2);
        }
        g.drawLine(1,1,30,30);//划线，两点确定一条直线
        //3.将内存中的图片(验证码)输出到页面展示
        ImageIO.write(image,"jpg",response.getOutputStream());
    }
    protected void doGet(HttpServletRequest request,HttpServletResponse response)
            throws ServletException,IOException{
        this.doPost(request,response);
    }
}
